from flask import g

'''
自定义一个装饰器，验证某些请求是否已经登陆过了，如果已经登陆了继续访问
本质上就是一个登录拦截 
'''


def login_required(func):
    def wrapper(*args,**kwargs):
        if g.user_id is not None: # 已经登陆过了
            return func(*args,**kwargs)
        else:
            return {'message':'Invalid token.'},400
    return wrapper